package cn.itcast.tanhua.sso.service;

import cn.hutool.core.util.ObjectUtil;
import cn.itcast.tanhua.common.mapper.UserMapper;
import cn.itcast.tanhua.common.pojo.User;
import cn.itcast.tanhua.dubbo.server.api.HuanXinApi;
import com.alibaba.dubbo.config.annotation.Reference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import io.jsonwebtoken.*;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * @ClassName UserService
 * @Description 用户登录
 * @Author WU
 * @Date 2021/5/24 22:17
 * @Version 1.0
 */
@Service
@Slf4j
public class UserService {
    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Reference(version = "1.0.0")
    private HuanXinApi huanXinApi;
    /**
    * @Description: 用户登录
    * @Param: phone、code
    * @return:
    * @Author: Mr.Tan
    * @Date: 2021/5/24
    */
    public String login(String phone, String code) {
        String redisKey = "CHECK_CODE_" + phone;
        boolean isNew = false;
//        校验验证码
        String redisData =  this.redisTemplate.opsForValue().get(redisKey);
//        如果接收到的验证码和redis中存的不一致，给出提示信息
        if (!StringUtils.equals(code, redisData)) {
            return null;
        }

//        验证码校验完成后，需要删除redis中的该条验证码
        this.redisTemplate.delete(redisKey);
//        通过手机号查找数据库，判断此用户是否为新用户
//        mybatis-plus条件查询
        QueryWrapper<User> wrapper = new QueryWrapper<>();
        wrapper.eq("mobile",phone);
        User user = this.userMapper.selectOne(wrapper);

//        查询到为空值，则为新用户，需要注册
        if (null == user) {
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));
//            注册
            this.userMapper.insert(user);
            isNew=true;
            //将该用户信息注册到环信平台
            Boolean register = this.huanXinApi.register(user.getId());
            if (!register) {
                log.error("注册到环信平台失败！ " + user.getId());
            }

        }

//        生成token，仅将id值放入
        Map<String,Object> claims=new HashMap<String, Object>();
        claims.put("id",user.getId());
        String token = Jwts.builder()
                .setClaims(claims)//payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret)//设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(12).toDate())//设置过期时间，12小时后过期
                .compact();

        try {
//        将成功的信息发送到MQ
            Map<String,Object> msg=new HashMap<>();
            msg.put("id",user.getId());
            msg.put("date",System.currentTimeMillis());

            this.rocketMQTemplate.convertAndSend("tanhua-sso-login", msg);
        } catch (MessagingException e) {
//            发送失败则写入日志
            log.error("发送消息失败！", e);
        }
        return token + "|" +isNew;

    }

    public User queryUserByToken(String token) {
        try {
//        通过token解析数据
            Map<String,Object> body =Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            User user = new User();
            user.setId(Long.valueOf(body.get("id").toString()));

            //需要返回user对象中的mobile，需要查询数据库获取到mobile数据
            //如果每次都查询数据库，必然会导致性能问题，需要对用户的手机号进行缓存操作
            //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
            //如果用户修改了手机号，需要同步修改redis中的数据

//        定义redis中的key
            String redisKey="TANHUA_USER_MOBILE_"+user.getId();

//        如果redis中有这个key，则拿出来
            if (this.redisTemplate.hasKey(redisKey)) {
                String mobile = this.redisTemplate.opsForValue().get(redisKey);
//                获取到手机号后放入user返回
                user.setMobile(mobile);
            }else {
    //            redis中没有则查询数据库，查询到信息后将手机号放入user返回，并将手机号放入redis
                User u = this.userMapper.selectById(user.getId());
                user.setMobile(u.getMobile());
                //将手机号写入到redis中
                //在jwt中的过期时间的单位为：秒
                long timeout = Long.valueOf(body.get("exp").toString()) * 1000 - System.currentTimeMillis();
                this.redisTemplate.opsForValue().set(redisKey, u.getMobile(), timeout, TimeUnit.MILLISECONDS);
            }
            return user;
        } catch (ExpiredJwtException e) {
            log.info("token已经过期！ token = " + token);
        } catch (Exception e) {
            log.error("token不合法！ token = "+ token, e);
        }
        return null;
    }

    public Boolean updatePhone(Long userId, String newPhone) {
        //先查询新手机号是否已经注册，如果已经注册，就不能修改
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", newPhone);
        User user = this.userMapper.selectOne(queryWrapper);
        if(ObjectUtil.isNotEmpty(user)){
            //新手机号已经被注册
            return false;
        }

        user = new User();
        user.setId(userId);
        user.setMobile(newPhone);

        return this.userMapper.updateById(user) > 0;
    }
}
